package com.wxgo.utils;

import javax.imageio.ImageIO;
import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics2D;
import java.awt.Image;
import java.awt.Toolkit;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.net.URL;

public class ImageUtils {

	private Font font = new Font("黑体", Font.BOLD, 20);// 添加字体的属性设置
	private Graphics2D g = null;
	private int fontsize = 12;
	private int x = 0;
	private int y = 0;

	/**
	 * 
	 * 导入本地图片到缓冲区
	 * 
	 */
	public BufferedImage loadImageLocal(String imgName) {
		Image src = Toolkit.getDefaultToolkit().getImage(imgName);
		// Image to BufferedImage
		return BufferedImageBuilder.toBufferedImage(src);
		// return ImageIO.read(new File(imgName));
	}

	/**
	 * 
	 * 导入网络图片到缓冲区
	 * 
	 */

	public BufferedImage loadImageUrl(String imgName) {
		try {
			URL url = new URL(imgName);
			return ImageIO.read(url);
		} catch (IOException e) {
			System.out.println(e.getMessage());
		}
		return null;
	}

	/**
	 * 
	 * 生成新图片到本地
	 * 
	 */
	public void writeImageLocal(String newImage, BufferedImage img) {
		if (newImage != null && img != null) {
			try {
				File outputfile = new File(newImage);
				ImageIO.write(img, "jpg", outputfile);
			} catch (IOException e) {
				System.out.println(e.getMessage());
			}
		}
	}

	/**
	 * 
	 * 修改图片,返回修改后的图片缓冲区（只输出一行文本）
	 * 
	 */
	public BufferedImage modifyImage(BufferedImage img, Object content, int x, int y) {
		try {
			int w = img.getWidth();
			int h = img.getHeight();
			g = img.createGraphics();
			g.setBackground(Color.WHITE);
			g.setColor(Color.RED);
			if (this.font != null)
				g.setFont(this.font);
			// 验证输出位置的纵坐标和横坐标
			if (x >= h || y >= w) {
				this.x = h - this.fontsize + 2;
				this.y = w;
			} else {
				this.x = x;
				this.y = y;
			}
			if (content != null) {
				g.drawString(content.toString(), this.x, this.y);
			}
			g.dispose();
		} catch (Exception e) {
			System.out.println(e.getMessage());
		}
		return img;
	}

	/**
	 * 
	 * 修改图片,返回修改后的图片缓冲区（输出多个文本段） xory：true表示将内容在一行中输出；false表示将内容多行输出
	 * 
	 */

	public BufferedImage modifyImage(BufferedImage img, Object[] contentArr, int x, int y, boolean xory) {
		try {
			int w = img.getWidth();
			int h = img.getHeight();
			g = img.createGraphics();
			g.setBackground(Color.WHITE);
			g.setColor(Color.RED);
			if (this.font != null)
				g.setFont(this.font);
			// 验证输出位置的纵坐标和横坐标

			if (x >= h || y >= w) {
				this.x = h - this.fontsize + 2;
				this.y = w;
			} else {
				this.x = x;
				this.y = y;
			}

			if (contentArr != null) {
				int arrlen = contentArr.length;
				if (xory) {
					for (int i = 0; i < arrlen; i++) {
						g.drawString(contentArr[i].toString(), this.x, this.y);
						this.x += contentArr[i].toString().length() * this.fontsize / 2 + 5;// 重新计算文本输出位置
					}
				} else {
					for (int i = 0; i < arrlen; i++) {
						g.drawString(contentArr[i].toString(), this.x, this.y);
						this.y += this.fontsize + 2;// 重新计算文本输出位置
					}
				}
			}
			g.dispose();
		} catch (Exception e) {
			System.out.println(e.getMessage());
		}
		return img;
	}

	/**
	 * 
	 * 修改图片,返回修改后的图片缓冲区（只输出一行文本）
	 *
	 * 
	 * 
	 * 时间:2007-10-8
	 *
	 * 
	 * 
	 * @param img
	 * 
	 * @return
	 * 
	 */

	public BufferedImage modifyImageYe(BufferedImage img) {
		try {
			int w = img.getWidth();
			int h = img.getHeight();
			g = img.createGraphics();
			g.setBackground(Color.WHITE);
			g.setColor(Color.RED);
			if (this.font != null)
				g.setFont(this.font);
			g.drawString("www.hi.baidu.com?xia_mingjian", w - 85, h - 5);
			g.dispose();
		} catch (Exception e) {
			System.out.println(e.getMessage());
		}
		return img;
	}

	public BufferedImage modifyImagetogeter(BufferedImage b, BufferedImage d) {
		try {
			int w = b.getWidth();
			int h = b.getHeight();
			g = d.createGraphics();
			g.drawImage(b, 100, 10, w, h, null);
			g.dispose();
		} catch (Exception e) {
			System.out.println(e.getMessage());
		}
		return d;

	}

	/**
	 * 
	 * 设定文字的字体等
	 * 
	 */
	public void setFont(String fontName, int fontStyle, int fontSize) {
		this.fontsize = fontSize;
		this.font = new Font(fontName, fontStyle, fontSize);
	}

	/**
	 * 得到图片画笔
	 */
	public Graphics2D getGraphics2D() {
		return g;
	}

	/**
	 * 设置图片画笔
	 */
	public void setGraphics2D(Graphics2D g) {
		this.g = g;
	}

	/**
	 * 程序启动入口
	 */
	public static void main(String[] args) {
		ImageUtils tt = new ImageUtils();
		BufferedImage b = tt.loadImageLocal("C:\\Users\\Administrator\\Desktop\\KumSoHyun\\KimSoHyun5_bak.jpg");
		b = tt.modifyImage(b, "第一次我说爱你的时候", 0, 100);
		tt.writeImageLocal("C:\\Users\\Administrator\\Desktop\\KumSoHyun\\KimSoHyun5_create.jpg", b);
	}

}
